SQUARE IN PYTHON